from .openai import call_openai
from .wenxin import call_wenxin
from .qianfan import call_qianfan


async def call_model(model_name: str, message: str):
    """
    根据模型名称调用对应模型
    """
    if model_name == "openai":
        return await call_openai(message)
    elif model_name == "wenxin":
        return await call_wenxin(message)
    elif model_name == "Qianfan-Agent-Speed-32K":
        return await call_qianfan(message)
    else:
        return "模型未找到"
